import matplotlib.pyplot as plt
import numpy as np

# Load the data
path = np.load("path_dir/conflict_path.npy", allow_pickle=True).tolist()
agents_list = np.load("path_dir/conflict_agents_list.npy", allow_pickle=True)
print(path)
# Plot the path
plt.figure(figsize=(10, 6))
for idx, agent_path in enumerate(path.values()):
    if idx == 5:
        plt.plot(agent_path[:, 0], agent_path[:, 1] + 20, label=f"Path{idx}")
        continue
    plt.plot(agent_path[:, 0], agent_path[:, 1], label=f"Path{idx}")

# Customize the plot
plt.title("Conflict Path Visualization")
plt.xlabel("X Coordinate")
plt.ylabel("Y Coordinate")
plt.legend()
plt.grid(True)

# Show the plot
plt.show()
